/*
 * Copyright 2013-2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.redis.demo.demos.web;

import com.redis.demo.common.utils.RedisUtils;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import java.util.concurrent.TimeUnit;

/**
 * @author <a href="mailto:chenxilzx1@gmail.com">theonefx</a>
 */
@Controller
public class BasicController {

    private final RedisUtils redisUtils;

    private final RedissonClient redissonClient;

    public BasicController(RedisUtils redisUtils, RedissonClient redissonClient) {
        this.redisUtils = redisUtils;
        this.redissonClient = redissonClient;
    }

    @RequestMapping("/hello")
    @ResponseBody
    public String hello(@RequestParam(name = "name", defaultValue = "unknown user") String name) {
        // 获取当前的线程ID
        //long id = Thread.currentThread().getId();
        long id = 55;
        // 加锁
        if (redisUtils.getLock(String.valueOf(id))) {
            try {
                // 业务逻辑
                System.out.println("加锁：业务逻辑。。。。" + id);

                Thread.sleep(10000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                // 释放锁
                redisUtils.releaseLock(String.valueOf(id));
            }
            return "哈哈哈哈";
        } else {
            System.out.println("未获取到锁");
            return "未获取到锁";
        }
    }

    @RequestMapping("/tryLock")
    @ResponseBody
    public String tryLock() {
        String lockKey = "lockKey";
        RLock lock = redissonClient.getLock(lockKey);
        boolean isLocked = false;
        try {
            // 尝试加锁，阻塞等待10秒，上锁后10秒自动解锁
            isLocked = lock.tryLock(10, 60, TimeUnit.SECONDS);
            if (isLocked) {
                // 业务逻辑
                System.out.println("获取到锁，执行业务逻辑");
                // 模拟业务处理
                Thread.sleep(20000);
                return "获取到锁，执行业务逻辑";
            } else {
                System.out.println("未获取到锁");
                return "未获取到锁";
            }
        } catch (InterruptedException e) {
            return "加锁异常";
        } finally {
            if (isLocked && lock.isHeldByCurrentThread()) {
                lock.unlock();
                System.out.println("释放锁");
            }
        }
    }

    // http://127.0.0.1:8080/user
    @RequestMapping("/user")
    @ResponseBody
    public User user() {
        User user = new User();
        user.setName("theonefx");
        user.setAge(666);
        return user;
    }

    // http://127.0.0.1:8080/save_user?name=newName&age=11
    @RequestMapping("/save_user")
    @ResponseBody
    public String saveUser(User u) {
        return "user will save: name=" + u.getName() + ", age=" + u.getAge();
    }

    // http://127.0.0.1:8080/html
    @RequestMapping("/html")
    public String html() {
        return "index.html";
    }

    @ModelAttribute
    public void parseUser(@RequestParam(name = "name", defaultValue = "unknown user") String name
            , @RequestParam(name = "age", defaultValue = "12") Integer age, User user) {
        user.setName("zhangsan");
        user.setAge(18);
    }
}
